from pydantic import BaseModel

from app.models.studies import EnhancementType
from app.schemas.doctors import ModalitySchema


class MedicalStudyBase(BaseModel):
    enhancement: EnhancementType
    labor_intensity: float


class MedicalStudyCreateSchema(MedicalStudyBase):
    modality_code: str


class MedicalStudyUpdateSchema(BaseModel):
    id: int
    labor_intensity: float


class MedicalStudySchema(MedicalStudyBase):
    id: int
    modality: ModalitySchema
